# pip install zhipuai 请先在终端进行安装

import zhipuai

zhipuai.api_key = "41805a5412196acfdd9d63a905f9039a.Z0TWy2173NoP1ZkI"
response = zhipuai.model_api.sse_invoke(
    model="chatglm_lite",
    prompt= [{"role":"user","content":"你是一名擅长写言情小说的作者，请你根据下面的主角设定，构思一篇爱情小说情节。\n陈冉:一名年轻的女警察，她一直在胡同区工作，深谙当地的风士人情和黑暗面。她有着坚定的信仰和强烈的正义感，不断追求真相和公正，因此在同事和居民中享有很高的声誉。\n赵煜:一名神秘的年轻男子，他似乎与胡同区的一些犯罪事件有关，但具体身份不明。赵爆是一个独来独往的人，不爱与人交往，但与陈冉有着一些奇妙的联系，两人的缘分就此开始。"}],
    temperature= 0.9,
    top_p= 0.7,
    incremental=True
)

for event in response.events():
    if event.event == "add":
        print(event.data, end="")
    elif event.event == "error" or event.event == "interrupted":
        print(event.data, end="")
    elif event.event == "finish":
        print(event.data)
        print(event.meta, end="")
    else:
        print(event.data, end="")